# TODO time step calculation